import argparse
import os

from Config import root_path
from 征兵.main import main
from 战场.main import entry_func
from 运输.main import resource_translate

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='传输参数')
    parser.add_argument('-st', '--script_type', type=int, default=0, help='脚本类型')
    # 兵种类型
    parser.add_argument('-s', '--soldier_type', type=int, default=0, help='兵种类型')
    # 兵种数量
    parser.add_argument('-n', '--num', type=int, default=0, help='兵种数量')
    # 是否双倍
    parser.add_argument('-d', '--double', type=bool, default=False, help='是否双倍')

    # 资源参数
    # 粮食数量
    parser.add_argument('-f', '--food', type=int, default=0, help='粮食数量')
    # 木材数量
    parser.add_argument('-w', '--wood', type=int, default=0, help='木材数量')
    # 石料数量
    parser.add_argument('-m', '--mine', type=int, default=0, help='石料数量')
    # 铁锭数量
    parser.add_argument('-i', '--iro', type=int, default=0, help='铁锭数量')
    # 黄金数量
    parser.add_argument('-g', '--gold', type=int, default=0, help='黄金数量')

    args = parser.parse_args()

    if os.path.exists("{}/stop.txt".format(root_path)):
        os.remove("{}/stop.txt".format(root_path))

    # 获取脚本类型
    script_type = args.script_type
    if script_type == 0:  # 运输
        resource_translate(args.food, args.wood, args.mine, args.iro, args.gold,
                           translate_type=args.soldier_type, num=args.num)
    elif script_type == 1:  # 征兵
        main(soldier_type=args.soldier_type, num=args.num, double=args.double)
    elif script_type == 2:  # 战场
        entry_func()
